#!/bin/bash

# Locust Worker节点启动脚本
# 可以运行在任意服务器上，连接到Master节点

# 配置参数
MASTER_HOST="21.124.219.93"  # Master节点IP
MASTER_PORT="5557"           # Master节点端口
TARGET_HOST=""               # 目标服务器IP，启动时指定

# 检查参数
if [ $# -eq 0 ]; then
    echo "使用方法: $0 <目标服务器IP>"
    echo "示例: $0 21.28.125.24"
    echo "示例: $0 21.124.219.93"
    exit 1
fi

TARGET_HOST=$1

echo "启动Locust Worker节点..."
echo "Master节点: $MASTER_HOST:$MASTER_PORT"
echo "目标服务器: http://$TARGET_HOST:80"
echo "================================"

# 检查locust是否安装
if ! command -v locust &> /dev/null; then
    echo "错误: Locust未安装，请先安装: pip install locust"
    exit 1
fi

# 检查locustfile.py是否存在
if [ ! -f "locustfile.py" ]; then
    echo "错误: locustfile.py文件不存在"
    exit 1
fi

# 启动Worker节点
locust -f locustfile.py --worker --master-host=$MASTER_HOST --master-port=$MASTER_PORT --host=http://$TARGET_HOST:80

echo "Worker节点已启动并连接到Master"